package com.ruoyi.common.core.domain;

import com.ruoyi.common.constant.HttpStatus;
import com.ruoyi.common.translate.TranslateNameService;
import com.ruoyi.common.utils.StringUtils;
import org.springframework.beans.BeansException;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;
import org.springframework.http.ResponseEntity;
import org.springframework.stereotype.Component;

import java.util.HashMap;
import java.util.List;
import java.util.Objects;

/**
 * 操作消息提醒
 *
 * @author ruoyi
 */
@Component
public class AjaxResult extends HashMap<String, Object> implements ApplicationContextAware {
    private static final long serialVersionUID = 1L;

    /**
     * 状态码
     */
    public static final String CODE_TAG = "code";

    /**
     * 返回内容
     */
    public static final String MSG_TAG = "msg";

    /**
     * 数据对象
     */
    public static final String DATA_TAG = "data";

    /**
     * 初始化一个新创建的 AjaxResult 对象，使其表示一个空消息。
     */
    public AjaxResult() {
    }

    /**
     * 初始化一个新创建的 AjaxResult 对象
     *
     * @param code 状态码
     * @param msg  返回内容
     */
    public AjaxResult(int code, String msg) {
        super.put(CODE_TAG, code);
        super.put(MSG_TAG, msg);
    }

    /**
     * 初始化一个新创建的 AjaxResult 对象
     *
     * @param code 状态码
     * @param msg  返回内容
     * @param data 数据对象
     */
    public AjaxResult(int code, String msg, Object data) {
        super.put(CODE_TAG, code);
        super.put(MSG_TAG, msg);
        if (StringUtils.isNotNull(data)) {
            super.put(DATA_TAG, data);
        }
    }

    /**
     * 返回成功消息
     *
     * @return 成功消息
     */
    public static AjaxResult success() {
        return AjaxResult.success("操作成功");
    }

    /**
     * 返回成功数据
     *
     * @return 成功消息
     */
    public static AjaxResult success(Object data) {
        return AjaxResult.success("操作成功", data);
    }

    /**
     * 返回成功消息
     *
     * @param msg 返回内容
     * @return 成功消息
     */
    public static AjaxResult success(String msg) {
        return AjaxResult.success(msg, null);
    }

    /**
     * 返回成功消息
     *
     * @param msg  返回内容
     * @param data 数据对象
     * @return 成功消息
     */
    public static AjaxResult success(String msg, Object data) {
        return new AjaxResult(HttpStatus.SUCCESS, msg, data);
    }

    /**
     * 返回警告消息
     *
     * @param msg 返回内容
     * @return 警告消息
     */
    public static AjaxResult warn(String msg) {
        return AjaxResult.warn(msg, null);
    }

    /**
     * 返回警告消息
     *
     * @param msg  返回内容
     * @param data 数据对象
     * @return 警告消息
     */
    public static AjaxResult warn(String msg, Object data) {
        return new AjaxResult(HttpStatus.WARN, msg, data);
    }

    /**
     * 返回错误消息
     *
     * @return 错误消息
     */
    public static AjaxResult error() {
        return AjaxResult.error("操作失败");
    }

    /**
     * 返回错误消息
     *
     * @param msg 返回内容
     * @return 错误消息
     */
    public static AjaxResult error(String msg) {
        return AjaxResult.error(msg, null);
//        throw new RuntimeException(msg);
    }

    /**
     * 返回错误消息
     *
     * @param msg  返回内容
     * @param data 数据对象
     * @return 错误消息
     */
    public static AjaxResult error(String msg, Object data) {
        return new AjaxResult(HttpStatus.ERROR, msg, data);
    }

    /**
     * 返回错误消息
     *
     * @param code 状态码
     * @param msg  返回内容
     * @return 错误消息
     */
    public static AjaxResult error(int code, String msg) {
        return new AjaxResult(code, msg, null);
    }


    /**
     * 返回空消息
     *
     * @return 空消息
     */
    public static AjaxResult noContent() {
        return new AjaxResult(HttpStatus.NO_CONTENT, "数据不存在", null);
    }


    /**
     * 是否为成功消息
     *
     * @return 结果
     */
    public boolean isSuccess() {
        return Objects.equals(HttpStatus.SUCCESS, this.get(CODE_TAG));
    }

    /**
     * 是否为警告消息
     *
     * @return 结果
     */
    public boolean isWarn() {
        return Objects.equals(HttpStatus.WARN, this.get(CODE_TAG));
    }

    /**
     * 是否为错误消息
     *
     * @return 结果
     */
    public boolean isError() {
        return Objects.equals(HttpStatus.ERROR, this.get(CODE_TAG));
    }

    /**
     * 方便链式调用
     *
     * @param key   键
     * @param value 值
     * @return 数据对象
     */
    @Override
    public AjaxResult put(String key, Object value) {
        super.put(key, value);
        return this;
    }


    /// -------------------------------------- 数据翻译 ---------------------------------------
    ///
    private static ApplicationContext applicationContext;
    private static TranslateNameService translateNameService;

    @Override
    public void setApplicationContext(ApplicationContext context) throws BeansException {
        applicationContext = context;
        // 延迟加载，避免循环依赖
    }

    /**
     * 获取ID映射服务（延迟初始化）
     */
    /// WARNING: 不要轻易改名字， 对应 TransLateNameService
    private static TranslateNameService getTransLateNameService() {
        if (translateNameService == null && applicationContext != null) {
            translateNameService = applicationContext.getBean(TranslateNameService.class);
        }
        return translateNameService;
    }

    /**
     * 同步处理
     */
    public static <T> AjaxResult successWithTranslate(T data) {
        if (data == null) {
            return AjaxResult.success();
        }

        doTranslate(data, false);
        return AjaxResult.success(data);
    }

    public static <T> void doTranslate(T data, boolean isAsync) {
        TranslateNameService service = getTransLateNameService();

        if (service != null) {
            if (data instanceof List) {
                if (isAsync) {
                    service.processListAsync((List<?>) data);
                } else {
                    service.processList((List<?>) data);
                }
            } else {
                if (isAsync) {
                    service.processAsync(data);
                } else {
                    service.process(data);
                }
            }
        }
    }


    /**
     * 异步处理（不等待映射完成）
     */
    public static <T> AjaxResult successAsyncWithTranslate(T data) {
        if (data == null) {
            return AjaxResult.success();
        }

        doTranslate(data, true);
        return AjaxResult.success(data);
    }

    /**
     * 预加载常用数据
     */
    public static void preload(String type, List<String> ids) {
        TranslateNameService service = getTransLateNameService();
        if (service != null) {
            service.preload(type, ids);
        }
    }

    /**
     * 获取性能统计
     */
    public static Object getStats() {
        TranslateNameService service = getTransLateNameService();
        return service != null ? service.getStats() : null;
    }

    /**
     * 手动设置服务（用于测试）
     */
    public static void setService(TranslateNameService service) {
        translateNameService = service;
    }

    /**
     * 重置服务（用于测试）
     */
    public static void reset() {
        translateNameService = null;
    }



}

